Micron Document
Fox's Git Mirrors

Node / RFnexus / code.git / commits / a80f5cd

Commit a80f5cd532e4b9ac2355fe61c93f9847929491bb

🗀 Browse tree at this commit

Parents : d4b836f
Author : Ahmet Inan <inan@aicodix.de>
Date : 2024-03-07T18:13:15+01:00

avoid indirection when getting the metrics ordered

Changes

3 files changed, 38 insertions(+), 19 deletions(-)

M polar_helper.hh -1
M polar_list_decoder.hh +18 -8
M polar_parity_aided.hh +20 -10

Diff

diff --git a/polar_helper.hh b/polar_helper.hh
index 3103469..09539d3 100644
--- a/polar_helper.hh
+++ b/polar_helper.hh
@@ -7,7 +7,6 @@ Copyright 2020 Ahmet Inan <inan@aicodix.de>
#pragma once
#include "simd.hh"
-#include "sort.hh"
namespace CODE {

diff --git a/polar_list_decoder.hh b/polar_list_decoder.hh
index 6e450e7..9575681 100644
--- a/polar_list_decoder.hh
+++ b/polar_list_decoder.hh
@@ -52,23 +52,33 @@ struct PolarListNode<TYPE, 0>
static MAP rate1(PATH *metric, TYPE *message, MAP *maps, int *count, TYPE *hard, TYPE *soft)
{
TYPE sft = soft[1];
- SIMD<PATH, 2*TYPE::SIZE> fork;
+ PATH fork[2*TYPE::SIZE];
for (int k = 0; k < TYPE::SIZE; ++k)
- fork.v[2*k] = fork.v[2*k+1] = metric[k];
+ fork[2*k] = fork[2*k+1] = metric[k];
for (int k = 0; k < TYPE::SIZE; ++k)
if (sft.v[k] < 0)
- fork.v[2*k] -= sft.v[k];
+ fork[2*k] -= sft.v[k];
else
- fork.v[2*k+1] += sft.v[k];
- auto perm = vorder(fork);
+ fork[2*k+1] += sft.v[k];
+ int perm[2*TYPE::SIZE];
+ perm[0] = 0;
+ for (int i = 1, j; i < 2*TYPE::SIZE; ++i) {
+ PATH t = fork[i];
+ for (j = i; j > 0 && fork[j-1] > t; --j) {
+ fork[j] = fork[j-1];
+ perm[j] = perm[j-1];
+ }
+ fork[j] = t;
+ perm[j] = i;
+ }
for (int k = 0; k < TYPE::SIZE; ++k)
- metric[k] = fork.v[perm.v[k]];
+ metric[k] = fork[k];
MAP map;
for (int k = 0; k < TYPE::SIZE; ++k)
- map.v[k] = perm.v[k] >> 1;
+ map.v[k] = perm[k] >> 1;
TYPE hrd;
for (int k = 0; k < TYPE::SIZE; ++k)
- hrd.v[k] = 1 - 2 * (perm.v[k] & 1);
+ hrd.v[k] = 1 - 2 * (perm[k] & 1);
message[*count] = hrd;
maps[*count] = map;
++*count;

diff --git a/polar_parity_aided.hh b/polar_parity_aided.hh
index 356c3e6..3268d37 100644
--- a/polar_parity_aided.hh
+++ b/polar_parity_aided.hh
@@ -100,32 +100,42 @@ struct PolarParityNode<TYPE, 0>
static MAP rate1(PATH *metric, TYPE *message, MAP *maps, int *index, TYPE *hard, TYPE *soft, TYPE *parity, int *count, int stride)
{
TYPE sft = soft[1];
- SIMD<PATH, 2*TYPE::SIZE> fork;
+ PATH fork[2*TYPE::SIZE];
for (int k = 0; k < TYPE::SIZE; ++k)
- fork.v[2*k] = fork.v[2*k+1] = metric[k];
+ fork[2*k] = fork[2*k+1] = metric[k];
for (int k = 0; k < TYPE::SIZE; ++k)
if (sft.v[k] < 0)
- fork.v[2*k] -= sft.v[k];
+ fork[2*k] -= sft.v[k];
else
- fork.v[2*k+1] += sft.v[k];
+ fork[2*k+1] += sft.v[k];
if (*count == 0) {
TYPE chk = *parity;
*parity = PH::one();
for (int k = 0; k < TYPE::SIZE; ++k)
if (chk.v[k] < 0)
- fork.v[2*k] = 1000000;
+ fork[2*k] = 1000000;
else
- fork.v[2*k+1] = 1000000;
+ fork[2*k+1] = 1000000;
+ }
+ int perm[2*TYPE::SIZE];
+ perm[0] = 0;
+ for (int i = 1, j; i < 2*TYPE::SIZE; ++i) {
+ PATH t = fork[i];
+ for (j = i; j > 0 && fork[j-1] > t; --j) {
+ fork[j] = fork[j-1];
+ perm[j] = perm[j-1];
+ }
+ fork[j] = t;
+ perm[j] = i;
}
- auto perm = vorder(fork);
for (int k = 0; k < TYPE::SIZE; ++k)
- metric[k] = fork.v[perm.v[k]];
+ metric[k] = fork[k];
MAP map;
for (int k = 0; k < TYPE::SIZE; ++k)
- map.v[k] = perm.v[k] >> 1;
+ map.v[k] = perm[k] >> 1;
TYPE hrd;
for (int k = 0; k < TYPE::SIZE; ++k)
- hrd.v[k] = 1 - 2 * (perm.v[k] & 1);
+ hrd.v[k] = 1 - 2 * (perm[k] & 1);
if (*count) {
message[*index] = hrd;
*parity = PH::qmul(vshuf(*parity, map), hrd);

Served by rngit 1.4.1 - Generated in 0.03s